<HTML>
<BODY BGCOLOR="white">
<PRE>
<FONT color="green">001</FONT>    package cheesymock;<a name="line.1"></a>
<FONT color="green">002</FONT>    <a name="line.2"></a>
<FONT color="green">003</FONT>    import java.lang.reflect.*;<a name="line.3"></a>
<FONT color="green">004</FONT>    import java.util.*;<a name="line.4"></a>
<FONT color="green">005</FONT>    import static cheesymock.MethodUtil.*;<a name="line.5"></a>
<FONT color="green">006</FONT>    <a name="line.6"></a>
<FONT color="green">007</FONT>    /**<a name="line.7"></a>
<FONT color="green">008</FONT>     * Verifies that all methods on the delegate object has been<a name="line.8"></a>
<FONT color="green">009</FONT>     * called. Does not verify property access, though.<a name="line.9"></a>
<FONT color="green">010</FONT>     *<a name="line.10"></a>
<FONT color="green">011</FONT>     * @author fredrik<a name="line.11"></a>
<FONT color="green">012</FONT>     */<a name="line.12"></a>
<FONT color="green">013</FONT>    public class MandatoryInvocation extends Validator {<a name="line.13"></a>
<FONT color="green">014</FONT>            private Set&lt;Method&gt; invoked = new HashSet&lt;Method&gt;();<a name="line.14"></a>
<FONT color="green">015</FONT>            <a name="line.15"></a>
<FONT color="green">016</FONT>            @Override<a name="line.16"></a>
<FONT color="green">017</FONT>            public void invocation(Method method, Object[] arguments) {<a name="line.17"></a>
<FONT color="green">018</FONT>                    if (delegate != null) {<a name="line.18"></a>
<FONT color="green">019</FONT>                            try {<a name="line.19"></a>
<FONT color="green">020</FONT>                                    Method m = resolveMethod(method, argumentTypes(arguments), this.delegate);<a name="line.20"></a>
<FONT color="green">021</FONT>                                    invoked.add(m);<a name="line.21"></a>
<FONT color="green">022</FONT>                            }<a name="line.22"></a>
<FONT color="green">023</FONT>                            catch (NoSuchMethodException e) {<a name="line.23"></a>
<FONT color="green">024</FONT>                            }<a name="line.24"></a>
<FONT color="green">025</FONT>                            catch (Exception e) {<a name="line.25"></a>
<FONT color="green">026</FONT>                                    throw new RuntimeException(e);<a name="line.26"></a>
<FONT color="green">027</FONT>                            }<a name="line.27"></a>
<FONT color="green">028</FONT>                    }<a name="line.28"></a>
<FONT color="green">029</FONT>            }<a name="line.29"></a>
<FONT color="green">030</FONT>    <a name="line.30"></a>
<FONT color="green">031</FONT>            @Override<a name="line.31"></a>
<FONT color="green">032</FONT>            public void check() {<a name="line.32"></a>
<FONT color="green">033</FONT>                    if (delegate != null) {<a name="line.33"></a>
<FONT color="green">034</FONT>                            for(Method m: delegate.getClass().getDeclaredMethods()) {<a name="line.34"></a>
<FONT color="green">035</FONT>                                    if (!invoked.contains(m)) {<a name="line.35"></a>
<FONT color="green">036</FONT>                                            throw new IllegalStateException("Method " + m.getName() + <a name="line.36"></a>
<FONT color="green">037</FONT>                                                            " is declared in the delegate object, but did not get called " + <a name="line.37"></a>
<FONT color="green">038</FONT>                                                            " during execution.");<a name="line.38"></a>
<FONT color="green">039</FONT>                                    }<a name="line.39"></a>
<FONT color="green">040</FONT>                            }<a name="line.40"></a>
<FONT color="green">041</FONT>                    }<a name="line.41"></a>
<FONT color="green">042</FONT>            }<a name="line.42"></a>
<FONT color="green">043</FONT>    }<a name="line.43"></a>




























































</PRE>
</BODY>
</HTML>
